package com.faxon.leetcode.k;

import org.junit.Test;

import java.util.LinkedList;
import java.util.Queue;

/**
 * @author Faxon
 * @version 1.0
 * @date 2020-04-22 10:13
 * 104. 二叉树的最大深度
 * 给定一个二叉树，找出其最大深度。
 * 二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
 * 说明: 叶子节点是指没有子节点的节点。
 * 示例：
 * 给定二叉树 [3,9,20,null,null,15,7]，
 * 3
 * / \
 * 9  20
 * /  \
 * 15   7
 * 返回它的最大深度 3 。
 **/
public class LeetCode104 {
    private static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }

    /**
     * 递归求解
     * @param root
     * @return
     */
    public int maxDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        TreeNode left = root.left;
        TreeNode right = root.right;
        if (left == null && right == null) {
            return 1;
        }
        int lehtHigh = maxDepth(left);
        int rightHigh = maxDepth(right);
        return Math.max(lehtHigh, rightHigh) + 1;
    }

    /**
     * 迭代求解(广度优先搜索BFS)
     * @param root
     * @return
     */
    public int maxDepth2(TreeNode root) {
        if (root == null) {
            return 0;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        if(root.right!=null){
            queue.add(root.right);
        }
        if(root.left!=null){
            queue.add(root.left);
        }
        int hight = 1;
        while (!queue.isEmpty()){
            hight++;
            int len = queue.size();
            for (int i = 0; i < len; i++) {
                TreeNode poll = queue.poll();
                if(poll.right!=null){
                    queue.add(poll.right);
                }
                if(poll.left!=null){
                    queue.add(poll.left);
                }
            }
        }
        return hight;
    }
}
